package com.eliauk.demo.controller.read;

import com.alibaba.excel.EasyExcel;
import com.eliauk.demo.dao.DemoDAO;
import com.eliauk.demo.entity.read.DemoData;
import com.eliauk.demo.utils.excel.listener.Demo5DataListener;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;

@RestController
@RequestMapping("demo5")
public class demo5Controller {

    /**
     * 多行表头，录入时需要跳过表头数据
     * @param file
     * @return
     * @throws IOException
     */
    @PostMapping("import")
    public String importData(MultipartFile file) throws IOException {
        InputStream in = file.getInputStream();
        EasyExcel.read(in, DemoData.class,new Demo5DataListener(new DemoDAO()))
                .sheet()
                .headRowNumber(2)//这里表示有两行表头，比如像demo5.xlsx 第一行是表头，第二行是数据示例，我们不想录入第二行的示例数据，就可以这样设置
                .doRead();
        return "success";
    }

}

